Skip to content

Enable rocm-support#353

Open
luukkonenr wants to merge 43 commits intobigscience-workshop:mainfrom
TurkuNLP:main
Open

Enable rocm-support#353
luukkonenr wants to merge 43 commits intobigscience-workshop:mainfrom
TurkuNLP:main

Conversation

@luukkonenr
Copy link
Copy Markdown

Patch rocm-support to fused kernels from Microsoft/Megatron-DeepSpeed-fork.

luukkonenr and others added 30 commits October 7, 2022 10:24
Patch HIP-support to fused kernels from Microsoft/Megatron-DeepSpeed-fork

Path HIP-support to fused kernels from Microsoft/Megatron-DeepSpeed-fork

remove some local files
Co-authored-by: Thomas Wang <24695242+thomasw21@users.noreply.github.com>
Co-authored-by: thomasw21 <24695242+thomasw21@users.noreply.github.com>
Resolves DataLoader segfaults seen in some multinode runs with PP > 1.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet